REGISTRY = {}

from .basic_controller import BasicMAC

REGISTRY["basic_mac"] = BasicMAC

from .multi_head_controller import MultiHeadMAC

REGISTRY["multi_head_mac"] = MultiHeadMAC